#include "G4VUserPrimaryGeneratorAction.hh"     // Action动作类
#include "G4GeneralParticleSource.hh"           // GPS头文件
#include "G4ParticleGun.hh"                     // 粒子枪头文件
#include "globals.hh"

class G4Event;

class G4ImprovePrimaryGeneratorAction : public G4VUserPrimaryGeneratorAction{

public:
    // 构造函数
    G4ImprovePrimaryGeneratorAction();
    // 析构函数
    ~G4ImprovePrimaryGeneratorAction();

    // 事件生成函数
    virtual void GeneratePrimaries(G4Event* e);

    // 用于获取粒子发生器的函数
    const G4ParticleGun* GetGPS() const{return particleGun;};

private:
    // 创建粒子枪
    G4ParticleGun * particleGun;


};